import LLM


def choose_region_llm(rider_info, region_num, now_location, positive_reviews_num, negative_reviews_num):
    info = {"now_location": now_location, "region_num": region_num,
            "positive_reviews_num": positive_reviews_num, "negative_reviews_num": negative_reviews_num,
            "sex": rider_info["sex"], "diligent_level": rider_info["diligent_level"]}

    response = LLM.get_response_from_llm(4, info)
    print("llm商区选择结果为：" + response)
    return response


if __name__ == "__main__":
    # test
    region_num = [2, 3, 4]
    now_location = [51, 68]
    rider_info = {'sex': 1, "diligent_level": 1}
    choose_region_llm(rider_info, region_num, now_location, 10, 2)
